import os
import uuid
from src.model.PictureModel import PictureModel
from src.init.init_db import db
from src.model.UserModel import UserModel


class ResTool:
    @staticmethod
    def generate_image_id():
        """
        生成唯一ID，用于图像名
        """
        return str(uuid.uuid4())

    @staticmethod
    def check_path(path: str):
        """
        path: 文件路径
        检查文件路径是否存在，不存在则创建
        """
        if not os.path.exists(path):
            os.makedirs(path)

    # def check_database(self, index: str):
    #     """
    #     index: 图片在文章中的索引
    #     图片在文章中的索引是否存在
    #     """
    def save_article_pic(self, aid: str, purl: str, index: int):
        """
        aid: 文章id
        purl: 图片路径
        index: 图片索引
        保存文章中图片信息
        """
        print(f'aid {aid} purl {purl} index {index}')
        pic = PictureModel(aid=aid, purl=purl, index=index)
        db.session.add(pic)
        db.session.commit()
